"""remove_columns_test3

Revision ID: 0586897cb73c
Revises: 8baad6145929
Create Date: 2025-06-16 21:53:27.066706

"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision: str = '0586897cb73c'
down_revision: Union[str, None] = '8baad6145929'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    """Upgrade schema."""
    # SQLite doesn't support dropping columns directly, so we use a batch operation
    with op.batch_alter_table('trades', schema=None) as batch_op:
        batch_op.drop_column('test3')


def downgrade() -> None:
    """Downgrade schema."""
    # SQLite doesn't support adding columns directly in all cases, so we use a batch operation
    with op.batch_alter_table('trades', schema=None) as batch_op:
        batch_op.add_column(sa.Column('test3', sa.VARCHAR(), nullable=True))
